CUDA_VISIBLE_DEVICES=1 python main_500.py --config_file configs/config_transformer_moe_attn.yaml --seed 1